c31d32b4ddfd24a1279a52d1f3d5856a0a54a7bc,cdap-app-fabric/src/test/java/co/cask/cdap/internal/app/runtime/spark/SparkProgramRunnerTest.java,SparkProgramRunnerTest,testSparkWithPartitionedFileSet,#,184

Before Change


    Map<String, String> args = new HashMap<>();
    PartitionedFileSetArguments.setInputPartitionFilter(
      args, PartitionFilter.builder().addRangeCondition("x", "na", "nx").build());
    args = RuntimeArguments.addScope(Scope.DATASET, "pfs", args);
    args.put("input", "pfs");
    runProgram(app, SparkAppUsingFileSet.CharCountSpecification.class, args);

After Change


        }
      });

    Map<String, String> inputArgs = new HashMap<>();
    PartitionedFileSetArguments.setInputPartitionFilter(
      inputArgs, PartitionFilter.builder().addRangeCondition("x", "na", "nx").build());
    Map<String, String> outputArgs = new HashMap<>();
    final PartitionKey outputKey = PartitionKey.builder().addStringField("x", "xx").build();
    PartitionedFileSetArguments.setOutputPartitionKey(outputArgs, outputKey);
    Map<String, String> args = new HashMap<>();
    args.putAll(RuntimeArguments.addScope(Scope.DATASET, "pfs", inputArgs));
    args.putAll(RuntimeArguments.addScope(Scope.DATASET, "pfs", outputArgs));
    args.put("input", "pfs");
    args.put("output", "pfs");

    runProgram(app, SparkAppUsingFileSet.CharCountSpecification.class, args);